Improved basic UI

Added testing functionality and improved prompts.

Cleaned up comments.