cleaned up main test