Simple particle simulator in pygame