test and comments

Tests now work with the new API syntax introduced in #4