Removed hard-coding of number of molecules to be loaded