adding proper testing