Got problem set 3.
Got problem set 4.
Polish the PLA setup as a function. Now it is free and easy to reuse.