started unit testing
minor change