i changed a bit of the function. but i think it fulfill all the requirements.