Made the script more flexible by using a positional parameter