Added functionality for travis CI