Added basic setup for unit testing