My final assignment code