corrected initial errors in code
fixed original probs
Merge pull request #2 from jackyujieyang/master

fixed prof's errors