Added better figures