Added more to the script