Update CSS based on bjz's comments