Also forgot a var XD
Merge pull request #5 from dapetcu21/master

Also forgot a var XD