Add utility functions
Change prompt format