Setup for 4th assignment