Updated the program to be able to handle comments now.