Finished my code