added more context to be able to generate more