Add main, remove unused variable