Added some missing lines
Merge pull request #4 from Rishik999/patch-1

Added some missing lines