Add display code, cleanup