bug fix & add background color