code was fixed for passes the tests. Test was added.