Add main to project 2