Cleaned up tiny bit of code