I edited it so that you can choose witch size