[Code cleanup] Minor

Removed unused function and did some output formatting

Signed-off-by: Siddharth Goel <siddharth98391@gmail.com>