add comment and clean up test