got the first two tests passing