code changes for pymk