Fixed minor code styling