did some visual changes to the code