Finished all the auxiliary layers

Finished the implementation of all the non-AI related layers.