Added main and 2 classes