modifying example for use with metatrain data