Code golfed again to reduce lines (making it cryptic?)