added work for problems 1 and 2