Improve how code looks