Main class! almost there