Statement marking for cleaner code generation