Added second test and got it to pass