Assignment 2 - Higher order functions, Unit tests