Switched to gaussian, nicer. Needs better gradient