Had to move the stuff from controller to view to make the repaint