Some code here is from ChatGPT