Add authentication checks