Added some onClick logic