fix code styling for consistency