wrapped game into 'def main():'
added some tests