Add callback as a registration arg. Closes #1