Refactor Controller code