Add support for custom user model