Initial implementation, one passing test