third homework added
[] second problem not finished