Better integration tests and added custom management commands support