should handle multiple training inputs now