Fixing weight count + optim with free cell count
Merge pull request #21 from JoeTheDentist/optim/free_cell_count

Optim/free cell count