Added lab-4 code