changing up some controller stuff