Improved the code container style. Removed repetitive lines and replaced with erlang "end." statement
