https://www.vuinsider.com/2020/06/psy401-assignment-1-solution-spring-2020.html