Added controller file too