Added MainController tests