Small cleaning

- cleaned up