Edits to make it runnable.