Finishing my code

finished