Clean up

haven't tested yet