Updating for travis build