Front end work, added button and shake effect