Make program stronger
Merge pull request #1 from Bitnick2002/master

Modified some code