Added controller for test automation