modified main for project