slight formatting adjustment in code