Take snapshot using pygame