Add first test and get it passing