Mostly done, button not working correctly.