Let them own the prototype, they are paying for it after all (however little that might be). It shouldn't be hard for you to completely rewrite the prototype if you want to go forward with the idea yourself (after all you're talking about 20 hours of initial work, and chances are the rewrite will be much faster and of much higher quality).