expected solution 2