added a simple case
implementation seems to be working fine