Fix suntax error, clean code