Added more controller classes to correspond to each screen