Added a few tests - you gotta start somewhere ey?