Separate display responsibilities from main script