Cleaned up some buttons

Added control from part button