got everything working in controller