Update pixel_data methods for flexibility.

Necessary to allow reassigning training/testing data by class.