Added custom prompt, other amazing changes