adding unit testing