Added some super basic testing