Fixing bugs in convo layer, and added a unit test setup, so that layers can be unit tested