Adding back in unit testsgit st