Slightly more complex than I thought . Will have a fix later, got to study for a semester test tomorrow though!

Cheers