Wrote two lines of code and it works!!!