Add basic testing