started better unit testing