Now homework can be done in less than 5 lines of code