Cleaned up a little bit of code