Cleaned up the code. Thanks!