Remove debugging prints. Add get_data method for saving