Updated Code to remove comments when running