first version of problems 3 and 4