Added problem description, made output nicer looking