fixing some minor code styling