Lab 6 fix
Merge pull request #15 from Ash927/master

Lab 6 fix