close to done the code