Clean up model and Main.