Clean up a bit
2 files changed