add couple tests