Adds batch weight updates and checker for only temperature