Adds solution to 'make it modular'