Set to work with custom user class