Added second test and made it pass.