Gave HW #1 optional problem a shot (it's hard)