Great feedback, the code definitely needed to be more robust :)