Some features

Hello! You project is awesome. I think you need to add some useful features:

- Parameter names with low line and string as value `Convolution(in_channels=n_dim)`;
- Some "x N" symbol near module that define how many times it's repeated;
- Displaying a complex module name as a parameter.
