Added some if statements for my PS1