1. integrate torchViz open source library for backprop visualisation 2. custom modules that could be grouped saved and used again by the user