Polishing up scheduler, added check for custom.py where you set up your custom scheduler