bug fixed and small styling change