Updated ok() helper to accept any entity type