Update diet controller to function correctly. Use it as template