Added testing framework / some tests