Fixed unit tests and added ubuntu integration test