Code compression

Compressed code to 5 lines