Got logic implemented, just testing