finished if else, working on bonus