update solution 2.2