Added new test and implementation for returning a single number