Fixed bug, ready for Wednesdays challenge