add more testing cases for current functionality