clean up

cleaning up the code