Additional unit test added, controller logic modified