comments added + fix include in main