minor refactor
minor refactor

PhilipCastiglione/learning-machines,PhilipCastiglione/learning-machines,PhilipCastiglione/learning-machines,PhilipCastiglione/learning-machines,PhilipCastiglione/learning-machines