Add optional load logic