Cleaned up a bit in edit tool.

Improved organization in live model.

Signed-off-by: Jakob Brandsborg Ehmsen <jakobehmsen.github@gmail.com>