Fixed error in line 21 & 24