started writing training wrapper