Simple stuff using python and pygame library