Building ResNet - Putting it all together