Cleaning the main.ml code.