added couple tests