finalizing my code

finalizing my code