fixing tests
adding assert
fixing comment
adding before method  tag
Merge pull request #280 from raghavgautam/fixes

Test fixes