Moved function to top so that it is evaluated beforehand.
Added two test cases for classify function.