Added some test stuff

Will add stuff for while and def later.