Add Labs interface

Fix tests