Fixing lesson_4 from types tutorial