Fixed AssertionError. Used return instead of console.log()