Make rm function a little prettier

Fix #1