Corrected the error, reverted code to original with weight added
Merge pull request #9 from McAwesomesauce/patch-3

Corrected the error, reverted code to original with weight added