adding working tests