Shorten 2-01's code

There's no code like no code! :)