Add nicer graphics