2 tests passing. With a little hard coding.