Cleaned the code up.