Added fix to lab 4