Figured out what you needed!

Now with asserts