Moved Main class from tests directory