Added a little testing