Add auth logic