Basic code used to demo unit testing