Figured out how to use states!