Adding functionality for level 1. WIP