Some updates to my code - [STILL DOESN'T WORK!!]