Clean up code and added help message.