started adding proper tests