Add unittest framework