Added a few lines, thanks to @simsso