Corrected first task of second homework