Clean up

Signed-off-by: Faiz Kothari <faiz.off93@gmail.com>