Add assumptions to code comment