adds some proper testing