working version, no button functionality yet