Cleaning up my proof a little