Fixing issue with lecture 4