Made the test pass
Code from coding lecture