cleaned up code

[skip ci]