Little maintenance

Fixing up rough edges