Ah yea forgot to add that back in. Edited the above code.