made changes to layout and added comments