Here is my code in main