* Polishing the embedding layer and the fc layer documentation
* Addressing code review feedback