Cleaned code up.