Used pattern matching to pass test for 1, added test for addition of two numbers.
Refactored to use simpler if/else construct.