Need to work on problems 1 and 4