Fix to alanine example.

Got overzealous in adding n_atoms to stuff.