Updated README to include Travis CI badge
Updated the model