Clean up

Signed-off-by: iceandfire <arhamjamal@gmail.com>