Changes to logic has been made.

1. Made changes to test case