Adding tests back into Travis