Clean up lejos code