Updated size to take arguments from the command line.