small changes to main.ml