Added a basic CNN + the code to test it on all the problems.