Added nicer figures