Update to include new styles