Clarified the code a bit here