Improve size.sh to accept arguments