Add unit testing for calculator