Fixed file structure and added comment