addition to main.r