Cleaned up a few things, added a prompt