adding controller, unit testing