work in progress - unit testing