Added new code to the script and changed up the constructor