Passed test by returning 1
Added new test for number 2