small aesthetic code changes