Improve coding style of proof mode.