complete the code (is now working)