Adds unit testing