Solution for 2.1