Removed MongoDB dependency for training