Tests for nesterov momentum, style fixes, docs