Adding travis badge and update tests