modified main.cs