remove unwanted files
make it work for all difficulty levels
more tidy