State information showing on UI correctly. Now working on buttons