Add test unit for new features #33