Added in number of guests prompt, cleaned up code