Assingment 5 nearly done. Annoying bug tho