Added controller in it's own seperate file