Fixed issue, code now works!