**Is your feature request related to a problem? Please describe.** Zairachem cannot run with about less than 60 molecules **Describe the solution you'd like** Remove some steps for small datasets. **Describe alternatives you've considered** Add a requirement of the minimum number of molecules to train a Zairachem model.