Begin to Fix Prime Function, all by lasat 8 test cases pass