Add support for custom user models