Add keyboard control to change mass