Cleaning up

Signed-off-by: Simarpreet Singh <simar@linux.com>