fix code better, colors and font sizes