Making code a bit more rubystic