Need help with proper class placement?