small changes - nearly ready, jest test