Add watch support for main files