Adding junit so some tests can be created