Working, need to add images and fix buttons