Main now works for both games