added import statement for local pygame variables. added screenshot