Added test to match the javascript exercise