Add unit tests capability