passed 1st test.
Fixing #5. test case