Modified the style accordning to HW4