Tests for nesterov momentum, style fixes