fixed my bug - moved to model and not controller