Added 'test' code