Cleanup, bug fixes.

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