still working on passing a few tests