Implemented exercise 2, fixed bug in 3D cell container