Added missing package dependencies, and added back support for convolution layers. Unit tests are passing.