Update to use existing sigmoid func