Update: Small change in coding with the CS