small code prettiness changes