added stuff shown in IntelliJ video