Add more update_fields