Add first few tests