fixing play screen code