Fix mistakes solution 2_4