Fixes broken button; closes #3