fixed code styling