Got unit testing working and wrote two tests