Bug fixes, cleanup.

Signed-off-by: FellowTraveler <F3llowTraveler@gmail.com>