little cleanup
omg it works