Fix proof bug and formatting bug