Add more styles for code