Python Prime or Composite Number Program