Added comments, small update to main.ml