fixing main class
First integration test