Almost there, working program tho