Been working on this problem set for the past few hours and I thought I was doing pretty well, but when I ran it, doesn\'t really work as expected. Kinda stuck at the moment
