looks like this is done though I chose a different implementation than the official solution.