Added framework for unit testing to main class