add working tests