Make register api more intuitive