modify controller
Merge pull request #19 from priyankalalchandani/feature/changecontroller

modify controller